'''
最终效果1
'''
# 导入海龟库
import turtle
from turtle import *


def rotation(edge_colors, edge):
    '''
    按照指定的边和颜色集合，旋转画图
    :param edge_colors:     颜色集合
    :param edge:            边的数量
    :return:
    '''
    angle = 360 / edge - 1
    color_size = len(edge_colors)
    for x in range(500):
        pencolor(edge_colors[x % color_size])
        forward(x)
        left(angle)
    return

# 主程序
turtle.tracer(0)         # 关闭画图过程
turtle.bgcolor("black")     # 黑色背景
colors = ['purple', 'yellow', 'blue', 'red','black','pink']     # 线的颜色集合
edge_size = numinput('edge size', 'please input edge size:')    # 用户输入边的数量
rotation(colors, edge_size) # 画图
turtle.done()   #画图结束，窗口保留
